package com.boat.ai.rag.embedding;

import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.model.embedding.onnx.allminilml6v2.AllMiniLmL6V2EmbeddingModel;
import dev.langchain4j.store.embedding.qdrant.QdrantEmbeddingStore;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.inject.Produces;

@ApplicationScoped
public class EmbeddingProvider {
    @Produces
    public QdrantEmbeddingStore store() {
        return new QdrantEmbeddingStore("kw_segment", "192.168.130.150", 6334, false, "text", "1234567890");
    }

    @Produces
    public EmbeddingModel model() {
        return new AllMiniLmL6V2EmbeddingModel();
    }
}
